#pragma once
#include "vtkmrmlmodeldisplaynode.h"
#include <vtkPolyData.h>

class vtkMRMLMyModelDisplayNode :
	public vtkMRMLModelDisplayNode
{
public:
	static vtkMRMLMyModelDisplayNode * New();
	vtkTypeRevisionMacro(vtkMRMLMyModelDisplayNode,vtkMRMLModelDisplayNode);
	virtual vtkMRMLNode* CreateNodeInstance();
	virtual const char* GetNodeTagName() {return "MyModelDisplayNode";};
	virtual vtkPolyData * GetPolyData();
	vtkSetObjectMacro(OutputPolyData, vtkPolyData);
	vtkSetObjectMacro(PolyData, vtkPolyData);
	virtual void BuildPipleline();
protected:
	vtkMRMLMyModelDisplayNode(void);
	~vtkMRMLMyModelDisplayNode(void);
	vtkMRMLMyModelDisplayNode(const vtkMRMLMyModelDisplayNode &);
	void operator = (const vtkMRMLMyModelDisplayNode &);
	vtkPolyData * PolyData;
	vtkPolyData * OutputPolyData;
};
